safetensors==0.3.0
datasets==2.10.1
sentencepiece==0.1.97
torch==1.13.1
numpy==1.22.3
einops==0.6.1
transformers==4.29.2
accelerate==0.20.3
black==22.3.0
isort==5.10.1
# originally tested with transformers=4.28.0.dev0 which handles Llama correctly
# git+https://github.com/huggingface/transformers/archive/464d420775653885760e30d24d3703e14f4e8a14.zip
